import argparse
import datetime

import pytz

tz = pytz.timezone('Asia/Shanghai')
current_time = datetime.datetime.now(tz)

def parse_args(check=True):
    parser = argparse.ArgumentParser()
    parser.add_argument('--checkpoint_path', type=str)
    parser.add_argument('--output_dir', type=str)
    parser.add_argument('--dataset_train', type=str)
    parser.add_argument('--dataset_val', type=str)
    parser.add_argument('--embedding_dir', type=str)
    parser.add_argument('--batch_size', type=int, default=16)
    parser.add_argument('--max_steps', type=int, default=50000)
    parser.add_argument('--num_epochs', type=int, default=5)
    parser.add_argument('--seq_nums', type=int, default=30)
    parser.add_argument('--learning_rate', type=float, default=1e-3)
    parser.add_argument('--state_size', type=int, default=512)

    FLAGS, unparsed = parser.parse_known_args()
    for x in dir(FLAGS):
        print(getattr(FLAGS, x))
    return FLAGS, unparsed


if __name__ == '__main__':
    FLAGS, unparsed = parse_args()

    for x in dir(FLAGS):
        print(getattr(FLAGS, x))